
from .decoder import BaseDecoder
from .DeepLabV3 import DeepLabV3
import torch.nn as nn 

class UNetBase(nn.Module):
    def __init__(self, out_channels):
        super().__init__()
        pass 
        self.encoder = DeepLabV3()
        self.decoder = BaseDecoder(out_channels=out_channels)

    def forward(self, x):
        pass 
        
        skips = self.encoder(x)
        out = self.decoder(skips)

        return out